package com.zj.study.utils;

import com.jcraft.jsch.Channel;
import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;
import com.jcraft.jsch.SftpException;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.util.Properties;
import java.util.Vector;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.IOUtils;

/**
 * @author 赵赳
 * @since 2022/11/29 17:20
 */
@Slf4j
public class SFTPUtil {


  private ChannelSftp sftp;

  private Session session;
  /** SFTP 登录用户名*/
  private String username;
  /** SFTP 登录密码*/
  private String password;
  /** 私钥 */
  private String privateKey;
  /** SFTP 服务器地址IP地址*/
  private String host;
  /** SFTP 端口*/
  private int port;

  /**
   * 构造基于密码认证的sftp对象
   * @param username 用户名
   * @param password 密码
   * @param host host地址
   * @param port 端口号
   */
  public SFTPUtil(String username, String password, String host, int port) {
    this.username = username;
    this.password = password;
    this.host = host;
    this.port = port;
  }

  /**
   * 构造基于秘钥认证的sftp对象
   * @param username 用户名
   * @param host host地址
   * @param port 端口号
   * @param privateKey 私钥
   */
  public SFTPUtil(String username, String host, int port, String privateKey) {
    this.username = username;
    this.host = host;
    this.port = port;
    this.privateKey = privateKey;
  }

  public SFTPUtil(){}

  /**
   * 连接sftp服务器
   *
   */
  public void login(){
    try {
      JSch jsch = new JSch();
      if (privateKey != null) {
        jsch.addIdentity(privateKey);// 设置私钥
        log.info("sftp connect,path of private key file：{}" , privateKey);
      }
      log.info("sftp connect by host:{} username:{}",host,username);

      session = jsch.getSession(username, host, port);
      log.info("Session is build");
      if (password != null) {
        session.setPassword(password);
      }
      Properties config = new Properties();
      config.put("StrictHostKeyChecking", "no");

      session.setConfig(config);
      session.connect();
      log.info("Session is connected");

      Channel channel = session.openChannel("sftp");
      channel.connect();
      log.info("channel is connected");

      sftp = (ChannelSftp) channel;
      log.info(String.format("sftp server host:[%s] port:[%s] is connect successfull", host, port));
    } catch (JSchException e) {
      log.error("Cannot connect to specified sftp server : {}:{} \n Exception message is: {}", host,
          port, e.getMessage());
    }
  }

  /**
   * 关闭连接 server
   */
  public void logout(){
    if (sftp != null) {
      if (sftp.isConnected()) {
        sftp.disconnect();
        log.info("sftp is closed already");
      }
    }
    if (session != null) {
      if (session.isConnected()) {
        session.disconnect();
        log.info("sshSession is closed already");
      }
    }
  }

  /**
   * 将输入流的数据上传到sftp作为文件
   *
   * @param directory 上传到该目录
   * @param sftpFileName sftp端文件名
   * @param input 输入流
   */
  public void upload(String directory, String sftpFileName, InputStream input) throws SftpException {
    try {
      sftp.cd(directory);
    } catch (SftpException e) {
      log.warn("directory is not exist");
      sftp.mkdir(directory);
      sftp.cd(directory);
    }
    sftp.put(input, sftpFileName);
    log.info("file:{} is upload successful" , sftpFileName);
  }

  /**
   * 上传单个文件
   *
   * @param directory 上传到sftp目录
   * @param uploadFile 要上传的文件,包括路径
   */
  public void upload(String directory, String uploadFile) throws FileNotFoundException, SftpException{
    File file = new File(uploadFile);
    upload(directory, file.getName(), new FileInputStream(file));
  }

  /**
   * 将byte[]上传到sftp，作为文件。注意:从String生成byte[]是，要指定字符集。
   *
   * @param directory 上传到sftp目录
   * @param sftpFileName 文件在sftp端的命名
   * @param byteArr 要上传的字节数组
   */
  public void upload(String directory, String sftpFileName, byte[] byteArr) throws SftpException{
    upload(directory, sftpFileName, new ByteArrayInputStream(byteArr));
  }

  /**
   * 将字符串按照指定的字符编码上传到sftp
   *
   * @param directory 上传到sftp目录
   * @param sftpFileName 文件在sftp端的命名
   * @param dataStr 待上传的数据
   * @param charsetName sftp上的文件，按该字符编码保存
   */
  public void upload(String directory, String sftpFileName, String dataStr, String charsetName) throws UnsupportedEncodingException, SftpException{
    upload(directory, sftpFileName, new ByteArrayInputStream(dataStr.getBytes(charsetName)));
  }

  /**
   * 下载文件
   *
   * @param directory 下载目录
   * @param downloadFile 下载的文件
   * @param saveFile 存在本地的路径
   */
  public void download(String directory, String downloadFile, String saveFile) throws SftpException, FileNotFoundException{
    if (directory != null && !"".equals(directory)) {
      sftp.cd(directory);
    }
    File file = new File(saveFile);
    sftp.get(downloadFile, new FileOutputStream(file));
    log.info("file:{} is download successful" , downloadFile);
  }
  /**
   * 下载文件
   * @param directory 下载目录
   * @param downloadFile 下载的文件名
   * @return 字节数组
   */
  public byte[] download(String directory, String downloadFile) throws SftpException, IOException{
    if (directory != null && !"".equals(directory)) {
      sftp.cd(directory);
    }
    InputStream is = sftp.get(downloadFile);

    byte[] fileData = IOUtils.toByteArray(is);

    log.info("file:{} is download successful" , downloadFile);
    return fileData;
  }

  /**
   * 删除文件
   *
   * @param directory 要删除文件所在目录
   * @param deleteFile 要删除的文件
   */
  public void delete(String directory, String deleteFile) throws SftpException{
    sftp.cd(directory);
    sftp.rm(deleteFile);
  }

  /**
   * 列出目录下的文件
   *
   * @param directory 要列出的目录
   */
  public Vector<?> listFiles(String directory) throws SftpException {
    return sftp.ls(directory);
  }

  /**
   * 迭代删除文件夹
   * @param dirPath 文件夹路径
   */
  public static void deleteDir(String dirPath) {
    File file = new File(dirPath);
    if(!file.exists()){
      return;
    }
    if(file.isFile())
    {
      var delete = file.delete();
    }else {
      File[] files = file.listFiles();
      if(files == null) {
        var delete = file.delete();
      }else {
        for (int i = 0; i < files.length; i++) {
          deleteDir(files[i].getAbsolutePath());
        }
        var delete = file.delete();
      }
    }
  }



  public static void main(String[] args) throws SftpException, IOException, JSchException {
    SFTPUtil sftp = new SFTPUtil("szmcmc-user", "Mcmc#1918", "10.173.0.100", 30022);
    sftp.login();
    File file = new File("D:\\test.txt");
    InputStream is = new FileInputStream(file);
    sftp.upload("/upload/settle", "hello.txt", is);
    sftp.logout();
  }

}
